# coding=utf-8

import akshare as ak


def get_sh_index_akshare():
    """
    使用AkShare获取上证指数数据
    """
    # 获取上证指数日频率数据
    df = ak.stock_zh_index_daily(symbol="sh000001")
    print("上证指数历史数据：")
    print(df.head())

    return df


# 获取数据
sh_data = get_sh_index_akshare()

"""
上证指数历史数据：
         date    open    high     low   close  volume
0  1990-12-19   96.05   99.98   95.79   99.98  126000
1  1990-12-20  104.30  104.39   99.98  104.39   19700
2  1990-12-21  109.07  109.13  103.73  109.13    2800
3  1990-12-24  113.57  114.55  109.13  114.55    3200
4  1990-12-25  120.09  120.25  114.55  120.25    1500
"""


# import tushare as ts
# import pandas as pd
#
# # 设置token（需要先注册tushare账号获取token）
# # ts.set_token('你的token')
#
# # 初始化pro接口
# # pro = ts.pro_api()
# pro = ts.pro_api(token='bbc6f076aa3d2b063cb26376ad12ffd94b53864b42d2344b3aa2039f')
#
#
# # 获取上证指数日线数据
# def get_shanghai_index():
#     """
#     获取上证指数历史数据
#     """
#     # 方法1：使用通用接口
#     df = ts.get_hist_data('000001', start='2020-01-01')
#     print("上证指数历史数据：")
#     print(df.head())
#
#     return df
#
#
# # 获取数据
# sh_data = get_shanghai_index()